from autoencoder.models import ConvAE2d, ConvAE3d, ConvAE